solving graph problems in python